(Done) task2 done
addition: tests