Source code for COMS 228