Get flash msg working