Added two more tests (taken for the Toptal exam)