My take on a improved iOS interview assignment