Finished the tests, but I might want to check how I'm checking the flags.