Advances in UI and predictions model... (needs to refactor)