Code

minor adjustments