Modified using Ivan Baev's code.