Move focus logic around. Dunno why i used click, haha