Started working on main program with testing