- Cleaned up the code.