added feature importance logging