Added two plotting things in their own function