Add return to main function
Minor Bug Fix
Merge pull request #2 from aparres/rama

Add return to main function