Added missing includes
Merge pull request #40 from zsbzsb/incoming

Added missing includes