Minor changes

Code reverted. lul