Extract features function

small update