UI Updates

I've updated the UI so that it works better on android/iOS