Added more python test stuff