Minor change:
- cleaning code