touching up code