Seperated the UI elements from rest