Updated code will try to use piston/graphics