added more plotting