Code refinement

Hope it helps! :palm_tree: