Makes nice python graph
getting messy