minor fixes
fix #1
add this #include for completeness