Fixed #3, thanks for pointing @ankit8898