Main screen looking good, with category passing - now to add the buttons!