Added back in install reference and cleaned up code a little