Added some figures
Removed rho_pw as varParams
Bug fixes
Cleaned up view
Changed dt for optimization