Added a few changes, program not running correctly though...