ITTE-850 Created unit tests for the task. And added return types to some
of the class methods.

Signed-off-by: Simeon Iliev <sisabski23@gmail.com>