code clean up; added units to plots