Updated tests to work with beauty columns and added extra flag.