Created new controller method to handle return variables better