Add use statement where it's necessary