Fixing warning about signed/unsigned mismatch
Adding missing #include