- Cleaned up code
- Fixed typo