matplotlib angle plot edited