Add better visual test