CS249A assignment 1