Refactor code, on pay controller, TODO: connect session via model only