Add initial basic tests