Assignment 1 for CSE337