For ECE551 final project