Laser weapon upgrade - minor code update