small adjustments
possible fix for code output