Assignment 2 just need to add vehicle problem