Add Python 2 support

Blocking as of now, need to properly convert super() calls.