still working on the logic piece