Task 3 solution

Signed-off-by: Nikola Iliev <nikola.iliev93@gmail.com>