Added main guard, so one can import this module
Merge pull request #5 from jbzdak/fix/main-method

Added main guard, so one can import this module