- Fix unhandled exception
- Refactor code