Trying to do it, but not working properly yet. Getting closer