Initial solution for task 7, initialize featureMatrix in own method