Small addition to features