diff --git a/rb_ws/src/buggy/scripts/2d_sim/velocity_ui.py b/rb_ws/src/buggy/scripts/2d_sim/velocity_ui.py
index bed311f9..83357f7b 100644
--- a/rb_ws/src/buggy/scripts/2d_sim/velocity_ui.py
+++ b/rb_ws/src/buggy/scripts/2d_sim/velocity_ui.py
@@ -1,9 +1,13 @@
 #! /usr/bin/env python3
+import numpy as np
 import rospy
 import sys
 from controller_2d import Controller
+from std_msgs.msg import Float64
 import threading
+import math
 import tkinter as tk
+import keyboard
 
 class VelocityUI: