Added some styling and fixed bugs preventing the entity categories from displaying