adding vector class for better code