Rewrite test(you can remix to add some changes cuz why not)