added files to easily edit
fixed class choice