got code working again