v2 code clean up