Modified code in order to test other student's code