CS224W final project