Complete UI basic functionality. Small fixes.