CS224S, homework 4