Add missing features