Changed few things, trying out flask