Modified script some