wrote test for first task *embarressed*