more tweaks and changes. really need more gpus