Modification to get classification website to work