Code and associated dataset for thesis