small plotting additions and bug corrections