Minor update to features