Corrected the example and also dropping 0s from the data matrix