fixed comments from review: imports and final