adding tests and setup