Add other missing files to project

Should be working now :)