Adding more tests and starting documentation