Working on proof of concept with small data first since loading large data kills my computer