Add findByCode method