Fixed small bug, seems to be functional now