Adding more tests and testing more python versions