[Minor] Dockblock and use statements fixed