Add in initial test