Changes to make it look better on smaller screens

Also some changes to make it a bit easier to understand for those who
didnt make it...