Added a line to deal with NA's in data