nicer graphs with pyplot