Improved code + added tests