cs224n assignment 5 solutions