Python Matplotlib Change Plot Color Python Function Parameters