WIP on main and extract features