Fixed wheel issue and pointer allocation in main (segfault)