couple lines added