updated comments + removed print statement