ML assignment 6 - extract features