Updated my homework