minor changes - looks good!