Assignment 2: Found the solution to the last bit. Just need to finish implementation.
Assignment 2: Problem solved. Only presentation now left to implement.