Rework  this to work with tf2 (?)