Improved Atomic Wave Functions using a Functional Transformation