Added some missing includes.