Refactoring the code a little bit.