Assignment 1 for 02-201 / 02-601