Got basic tests working