CS224S, homework 3