Added tests

100% test coverage: ✔