Adding gpu mode