plots.py added + some code added and shuffled