Change the UI, added some snake logic