Better imgui integration

Naive but better imgui integration -> to improve in the future after inputs rework