add a more concise solution