Add categories to flashes.

Remove category for final if block in base
template.