Add D2 stuff and fix rest