only train once
fixed hmm
need to remove missing data