UI: Added red squares, fixed bugs