Minor feature update