Bringing code up to par with Teacher's code