Cleaned until line 189, testing if niche models work.