passed tests && added main guard