cleanup

tidying up visuals, utilised dataset