Add more documentation
adding more tests