Tightened up the code