Efficiency Update

Changed a small section of code to improve efficiency.