Add custom methods to classes