I will implement unit tests :)