UI update

- UI updated
- Checkpoint added
UI updated

- UI updated
- Checkpoint added