CS284r Final Project