Modified Main wrt Display Class