Some comments added and classificaiton steps logging slightly improved