Smarter approach to returning list, as pointed out by Dean