added a couple lines