Louis | fixes some UI issues