Add plotting.py
Import and use plotting.py