Made some changes. Added a method