Fixed some stuff. Added an await so that things work properly.