more tests and logic for correct answers