Added some fancy plot stuff