Undoes more of Daniel's changes

Moved the square root to the correct place. oops