Updated to new method to get features