Making a nice plotting function