Fixed bug on l. 15, accidentally used = insead of ==