Continuation of my cs287 final project