still trying to get the lemon dataset to work