Improve die script