Use models method not class var and remove __END__