Clean-up

Adding comment/ removing excess code.