Added changes I made at home