Data load and visualization taken from tutorial