Small edit-- clean up