code slight clean up