Add tests to all method

Close #5