Updated to work better

Todd Pickell @502380037 <todd.pickell@ge.com>