Updated the __main__.py to call module methods

Closes #80 (closed)

This MR should resolve that __main__.py calls methods instead of import <model> only.