Add chi squared calculation!