Improved with_flags