Some leftover changes from cs229.