Changed async to use await