Finished writing tests and updated documentation