Fixed the last bug. Functionality implemented. Forgot remove.