Done getting data and parsing, need to finish AI part