add tests back -- we will add them later