CS 838 Project Stage 4