Use TestTask, add tests