Add features back