Fixing Sockets and UI