Custom class according to the flash type