remove unnecessary imports + reformat code