Our awesome CS 229 project