cs401 assignment 1 solution