Stanford CS 229 coursework in python