Changed Program as per assignment

Changed Program as per assignment