Added training data
Added other cell types