Implement working selected word highlighting :)