adding tests and prettier output