UI design done, just hooking up the functions