Add missing include and pragma