Updated solution for bug fixing