Added additional needed line, clarified variable