added extra files and solved little issues