Fixed plot in Main.R