Remake of Flood app in Swift