added tests and fixed a helper function
added a test for a wheel straight