Code reformat & import optimisation