Refactoring and added screenshot saving part
Updated .readme