update with homework 5 for CS202 section