[TASK] Code Cleanup - #1