Added some code from benanne's kaggle-galaxies repo