Cleaned up code and tests, learning how to mock outputs in python and added some doccodes