Add initial attemp of unit testing