Add some styling and remove unnecessary stuff