Updated to fix tiny data set