Code for making astronomy plots with python