working, need to add animation and graphs still