Add links between heroku and github and tidy up code