refactored method
Removed my method, using the helper method, duhh
Added sort param name to button