Update - changed some UI formatting to better fit the screen