Added main() guard

So you can import this function in another one without executing the main() part.
Merge pull request #1 from m45t3r/patch-1

Added main() guard