Improved code. Got 100%