added assignment 1 for CS261