making it work - need to fix state