Update layout functionality in order to handle new GraphML format of IR1