Added UI tests to US501. Solved issue passing the message via ViewBag.