Added some debugging
Added missing header