Modified homework1 to work in the same manner as everyone else