Add gun logic