Added more functionality to the python libraries