Add initial tests