Updated feature preprocessing section