Made change suggested and closed #9.