Added some python support