Tried to add plotting