add missing use statements and protected properties