Finished unit tests

Maybe add documentation