Second homework for CS201