cleaned up tests and added return types
cleaned up and made refactors. Also working on readme