Two features enabled
- Data loading now is waited automatically.
- Gradient can be switched.