Forgot to add headers to main.cpp - fixed