updates, added more plotting