[TASK] Fix code formatting