react code refactor