Added code for GPU exercise