Add Part class to Models