Fixed imports and added square roots of 1