Replace helper with method in model