Moved helper flash_class to the right file