Adding changes to homework stub