missing header
minor changes