Updated Main class so that it works

Signed-off-by: callum <callumt47@hotmail.co.uk>