Fixed bug with screenshot and made console output