added couple lines