Adding in gun code