added some more tests and flash messages