if else main file and updated square types with output function overloaded