COS 301 Mini project part 2