Refactored the code.