Improved imgui integration. Colors still seem wrong!