Can you write me a program in Python for simulating nuclear reactor?