Add some UI logic
Add flash