Added homework code