I think that you can do it in one line