Finished my homework example