Add more tests and add more documentation in the different python modules