Add missing use statements