Split code into multiple files. Untested.
Fixed import
Merge fix
Merge pull request #8 from GradysGhost/code-split

Code split; fixes Issue #6.