Added code to compute feature importance and print it to file to be
printed later