add include header to imlementation
correct var type in state #17