Add some more test
Add my solution to the task