Added code to the main file
Merge pull request #3 from Sabuntu/master

Added code to the main file