Redoing most of project for cleaner code. Started entities file and added Tree class