changed default device to the GPU