Add few initial tests