I like @ElDeveloper's proposed code