Adding main file for working code