Minor Code refactor