attempt to add logic