Clean up code
Add eigenface logic