fixed visualization code to work with Romeo