Improved main function to give options