slight clean up