Add conditionals to controller for new methods