Add use note and tidy code.