Fixed according to Prof's comment