Alex fix && refacator