Cleaning up + Final plots

Added the final results to the plots function and sorted the plots out.