Add unitary tests

check methods to do what they should do and prevent errors