Gpu version sorta working