restructure for tests, add colors and prompt