Updating GPU implementation