Refactor to show only four results