Started working basic class functionality