Initial addition of python code.

Signed-off-by: Collin Bode <collin@berkeley.edu>