Trying flatten solution (#7)