Uncomment code. (still needs to be tested)